
from classeCarro import *

class Repositorio_Carro:
    def Cadastrar_Carro(self,carro):
        import MySQLdb
        db = MySQLdb.connect('localhost','root','root')
        db.select_db('locadora')
        cursor = db.cursor()
        sql = ('INSERT INTO carro (marca,modelo,ano,cor,placa) VALUES ("%s","%s","%s","%s","%s")'%(carro.getMarca(), carro.getModelo(), carro.getAno(),carro.getCor(),carro.getPlaca()))
        cursor.execute(sql)
        db.commit()
        db.close()


    def pesquisarCarro(self):
        import MySQLdb
        db = MySQLdb.connect('localhost','root','root')
        db.select_db('locadora')
        cursor = db.cursor()
        modelo = raw_input('digite o modelo que deseja pesquisar:\n').upper()
        sql = ('SELECT * FROM carro WHERE modelo = "%s"' %(modelo))
        cursor.execute(sql)
        lines = cursor.fetchall()
        if (lines == ()):
            print 'Titulo nao localizado!! Tente novamente.\n'.upper()

        else:
            for i in lines:
                marca = i[5]
                placa = i[1]
                modelo = i[2]
                ano = i[3]
                cor = i[4]
                print 'PLACA: %s - MODELO: %s - ANO: %s - COR: %s - MARCA: %s' %(placa,modelo,ano,cor,marca)
                

    def pesquisarCarro2(self,placa):
        import MySQLdb
        db = MySQLdb.connect('localhost','root','root')
        db.select_db('locadora')
        cursor = db.cursor()
        
        sql = ('SELECT * FROM carro WHERE placa = "%s"' %(placa))
        cursor.execute(sql)
        lines = cursor.fetchall()
        if (lines == ()):
            print 'Titulo nao localizado!! Tente novamente.\n'.upper()

        else:
            for i in lines:
                marca = i[5]
                placa = i[1]
                modelo = i[2]
                ano = i[3]
                cor = i[4]
                
                carro = Carro(marca,modelo,ano,cor,placa)
                return carro

    def refreshDb(self,carro):
        import MySQLdb
        db = MySQLdb.connect('localhost','root','root')
        db.select_db('locadora')
        cursor = db.cursor()
        sql = 'UPDATE carro SET modelo = "%s",marca = "%s",ano = "%s",cor = "%s" WHERE placa = "%s"' %(carro.getModelo(),carro.getMarca(),carro.getAno(),carro.getCor(),carro.getPlaca())
        if (cursor.execute(sql)):
            db.commit()
            db.close()
        else:
            print "erro"

        


    def excluirCarro(self,carro):
        import MySQLdb
        db = MySQLdb.connect('localhost','root','root')
        db.select_db('locadora')
        cursor = db.cursor()

        
        sql = ('delete FROM carro WHERE placa = "%s"' %(carro.getPlaca()))
        
         
        if (cursor.execute(sql)):
            
            db.commit()
            print 'Carro excluido com sucesso!!!'.upper()
        else:
            print "erro"
            #db.rollback()
            #db.close()

    def editarTitulo(self):
        import MySQLdb
        db = MySQLdb.connect('localhost','root','root')
        db.select_db('locadora')
        cursor = db.cursor()
        acervo = Repositorio_DVD()
        acervo.pesquisarTitulo()
        print ('Se o titulo nao for localizado, digite 0 e depois <Enter>\n'.upper())
        coddvd = raw_input('Informe o Cod do DVD que deseja alterar:\n'.upper())
        if (coddvd == 0):
            print 'Titulo nao localizado no acervo'.upper()
        else:
            novoTitulo = raw_input('Informe o novo nome que deseja atribuir ao titulo: \n'.upper()).upper()
            sql = 'UPDATE dvd SET titulo = "%s" WHERE coddvd = "%s"' %(novoTitulo,coddvd)
            cursor.execute(sql)
            db.commit()
            db.close()
            print 'Seus dados foram alterados com sucesso!!!'.upper()

    def editarGenero(self):
        import MySQLdb
        db = MySQLdb.connect('localhost','root','root')
        db.select_db('locadora')
        cursor = db.cursor()
        acervo = Repositorio_DVD()
        acervo.pesquisarTitulo()
        print ('Se o titulo nao for localizado, digite 0 e depois <ENTER>'.upper())
        coddvd = raw_input('Informe o Cod do DVD que deseja alterar:\n'.upper())
        if (coddvd == 0):
            print 'Titulo nao localizado no acervo'.upper()
        else:
            novoGenero = raw_input('Informe o novo genero que deseja atribuir: \n'.upper()).upper()
            sql = 'UPDATE dvd SET genero = "%s" WHERE coddvd = "%s"' %(novoGenero,coddvd)
            cursor.execute(sql)
            db.commit()
            db.close()
            print 'Seus dados foram alterados com sucesso!!!'.upper()

    def editarQuantidade(self):
        import MySQLdb
        db = MySQLdb.connect('localhost','root','root')
        db.select_db('locadora')
        cursor = db.cursor()
        acervo = Repositorio_DVD()
        acervo.pesquisarTitulo()
        print ('Se o titulo nao for localizado, digite 0 e depois <ENTER>:\n'.upper())
        coddvd = raw_input('Informe o Cod do DVD que deseja alterar:\n'.upper())
        if (coddvd == 0):
            print 'Titulo nao localizado!!'.upper()
        else:
            novaQuanti = raw_input('Informe a nova quantidade que deseja atribuir:\n'.upper()).upper()
            sql = 'UPDATE dvd SET quantidade = "%s" WHERE coddvd = "%s"' %(novaQuanti,coddvd)
            cursor.execute(sql)
            db.commit()
            db.close()
            print 'Seus dados foram alterados com sucesso!!!'.upper()
            

    def pesquisarGenero(self):
        import MySQLdb
        db = MySQLdb.connect('localhost','root','root')
        db.select_db('locadora')
        cursor = db.cursor()
        acervo = Repositorio_DVD()
        acervo.pesquisarTitulo()
        titulo = raw_input('digite o genero que deseja pesquisar:\n'.upper()).upper()
        sql = ('SELECT * FROM dvd WHERE titulo = "%s"' %(genero))
        cursor.execute(sql)
        lines = cursor.fetchall()
        if (lines == ()):
            print 'genero nao localizado!! Tente novamente.\n'.upper()

        else:
            for i in lines:
                titulo = i[0]
                genero = i[1]
                quantidade = i[2]
                coddvd = i[3]
                print 'TITULO: %s - GENERO: %s - QUANTIDADE: %s - CODIGO DO DVD: %s' %(titulo,genero,quantidade,coddvd)
    
            
##
##repositorio = Repositorio_DVD()
####repositorio.Cadastrar_dvd()
##repositorio.pesquisarTitulo()
##repositorio.apagarTitulo()
##repositorio.editarTitulo()
##repositorio.editarGenero()
##repositorio.editarQuantidade()
